"""
使用stu数据库下的user表完成。
编写一个方法，完成注册功能，
要求注册的用户名不能与已有用户名相同
"""

import pymysql

def connect():
    kwargs = {
        "host": "localhost",
        "port": 3306,
        "user": "root",
        "password": "123456",
        "database": "stu",
        "charset": "utf8"
    }

    # 连接数据库
    db = pymysql.connect(**kwargs)

    # 创建游标:进行数据操作得到操作结果
    cur = db.cursor()
    return db, cur


def check(cur):
    # 数据读写
    name = input("请输入用户名：")
    passwd = input("请输入密码：")
    sql = "select user from user;"
    cur.execute(sql)
    result = list(cur.fetchall())
    for item in result:
        if name == item:
            print("用户名已存在，请重新输入")

    sql="insert into user (user,password) values (%s,%s);"
    cur.execute(sql,[name,passwd])
    print("注册成功")


def main():
    db, cur = connect()
    check(cur)
    cur.close()
    db.close()


if __name__ == '__main__':
    main()
